#!/usr/bin/python
# -*- coding: utf-8 -*-

#-------------------------------------------------
#Create At: 2008-12-15 下午10:29:10 
#Create By: Fla.sam
#
#DESC: 
#
#-------------------------------------------------

from CoreApp.models import Place
from django.contrib.auth.models import User, Group
from django.core.management.base import NoArgsCommand

AD_PLACE = ( 
            ( u'中国', None ), \
            ( u'北京市', u'中国' ), \
            ( u'上海市', u'中国' ), \
            ( u'天津市', u'中国' ), \
            ( u'重庆市', u'中国' ), \
            ( u'内蒙古', u'中国' ), \
            ( u'呼和浩特市', u'内蒙古' ), \
            ( u'河北省', u'中国' ), \
            ( u'石家庄市', u'河北省' ), \
            ( u'江西省', u'中国' ), \
            ( u'南昌市', u'江西省' ), \
            ( u'山东省', u'中国' ), \
            ( u'济南市', u'山东省' ), \
            ( u'山西省', u'中国' ), \
            ( u'太原市', u'山西省' ), \
            ( u'河南省', u'中国' ), \
            ( u'郑州市', u'河南省' ), \
            ( u'辽宁省', u'中国' ), \
            ( u'沈阳市', u'辽宁省' ), \
            ( u'湖北省', u'中国' ), \
            ( u'武汉市', u'湖北省' ), \
            ( u'吉林省', u'中国' ), \
            ( u'长春市', u'吉林省' ), \
            ( u'湖南省', u'中国' ), \
            ( u'长沙市', u'湖南省' ), \
            ( u'黑龙江省', u'中国' ), \
            ( u'哈尔滨市', u'黑龙江省' ), \
            ( u'广东省', u'中国' ), \
            ( u'广州市', u'广东省' ), \
            ( u'江苏省', u'中国' ), \
            ( u'南京市', u'江苏省' ), \
            ( u'广西省', u'中国' ), \
            ( u'南宁市', u'广西省' ), \
            ( u'海南省', u'中国' ), \
            ( u'海口市', u'海南省' ), \
            ( u'四川省', u'中国' ), \
            ( u'成都市', u'四川省' ), \
            ( u'浙江省', u'中国' ), \
            ( u'杭州市', u'浙江省' ), \
            ( u'贵州省', u'中国' ), \
            ( u'贵阳市', u'贵州省' ), \
            ( u'安徽省', u'中国' ), \
            ( u'合肥市', u'安徽省' ), \
            ( u'云南省', u'中国' ), \
            ( u'昆明市', u'云南省' ), \
            ( u'福建省', u'中国' ), \
            ( u'福州市', u'福建省' ), \
            ( u'陕西省', u'中国' ), \
            ( u'西安市', u'陕西省' ), \
            ( u'西藏自治区', u'中国' ), \
            ( u'拉萨市', u'西藏自治区' ), \
            ( u'甘肃省', u'中国' ), \
            ( u'兰州市', u'甘肃省' ), \
            ( u'青海省', u'中国' ), \
            ( u'西宁市', u'青海省' ), \
            ( u'宁夏回族自治区', u'中国' ), \
            ( u'银川市', u'宁夏回族自治区' ), \
            ( u'新疆维吾尔自治区', u'中国' ), \
            ( u'乌鲁木齐市', u'新疆维吾尔自治区' ), \
            )


def init_ad_place_table():
    for x in AD_PLACE:
        r = Place.objects.filter(place_name__exact=x[1]) #@UndefinedVariable
        if r.count() >= 1 :
            print u'Add Place %s to %s' % (x[0],x[1])
            ad_place = Place(place_name=x[0], place_parent=r[0])
            ad_place.save()
        else: 
            print u'Add Place %s' % x[0]
            ad_place = Place(place_name=x[0], place_parent=None)
            ad_place.save()

USER_TYPE_CHOISE = ( ( u'AD_ADMIN', u'系统管理员' ), \
                     ( u'AD_SMANAGER', u'超级管理员' ), \
                     ( u'AD_MANAGER', u'管理员' ), \
                     ( u'AD_AGENCY', u'广告代理商' ), \
                     ( u'AD_CUSTOMER', u'广告投放客户' ), \
                     ( u'AD_ENDUSER', u'广告終端客户' ), \
                     ( u'AD_ANONYMOUS', u'参观客户' )
                     )
            
def init_group_table():
    for x , y in USER_TYPE_CHOISE:
        print x,y
        u = x.split('_')[1].lower()
        g = Group(name=x)
        g.save()
        u = User(username=u, is_active=True)
        u.set_password(u)
        u.save()
        u.groups.add(g)
        
class Command( NoArgsCommand ):
    help = 'Init all database value'
    
    def handle_noargs( self, **options ):
        print 'Staring initialization ......'

        print 'Create Place items ......'
        init_ad_place_table()
        print 'Create Place item Done.'
        print 'Create Group ......'
        init_group_table()
        print 'Create Group Done.'
        
        
        
        
        